from tasks.deep_learning.models import LSTM_Model
if __name__ == '__main__':
    import os
    root_path = os.path.dirname(os.path.abspath(__file__))
    root_path = os.path.join(root_path,"dataset")
    from corpus.stock_net_dataset import StockNetDataset
    train_x,train_y = StockNetDataset.load_data(os.path.join(root_path,'GOOG-year.csv'))

    model = LSTM_Model()
    model.fit(train_x, train_y,epochs=30,batch_size=16)
